#ifndef TREE_MODEL_H_ICLUDED
#define TREE_MODEL_H_ICLUDED

#include <QAbstractItemModel>
#include <QStringList>
#include <QModelIndex>
#include <QVariant>



class TreeItem;


class TreeModel : public QAbstractItemModel
{
    Q_OBJECT

private:
    TreeItem* mRootItem;

public:
    explicit TreeModel(const QStringList& headers, QObject* parent = nullptr);
    TreeModel(const QStringList& headers, const QString& data,
              QObject* parent = nullptr);
    ~TreeModel();

    QVariant data(const QModelIndex& index, int role) const override;
    QVariant headerData(int section, Qt::Orientation orientation,
                        int role = Qt::DisplayRole) const override;

    bool setData(const QModelIndex& index, const QVariant& value,
                 int role = Qt::EditRole) override;
    bool setHeaderData(int section, Qt::Orientation orientation,
                       const QVariant& value, int role = Qt::EditRole) override;


    QModelIndex index(int row, int column,
                      const QModelIndex& parent = QModelIndex()) const override;

    QModelIndex parent(const QModelIndex& index) const override;

    Qt::ItemFlags flags(const QModelIndex& index) const override;


    int rowCount(const QModelIndex& parent = QModelIndex()) const override;
    int columnCount(const QModelIndex& parent = QModelIndex()) const override;


    bool insertRows(int position, int rows,
                    const QModelIndex& parent = QModelIndex()) override;
    bool removeRows(int position, int rows,
                    const QModelIndex& parent = QModelIndex()) override;

    bool insertColumns(int position, int columns,
                       const QModelIndex& parent = QModelIndex()) override;
    bool removeColumns(int position, int columns,
                       const QModelIndex& parent = QModelIndex()) override;



private:
    void setupModelData(const QStringList& lines, TreeItem* parent);
    TreeItem* getItem(const QModelIndex& index) const;
};


#endif // TREE_MODEL_H_ICLUDED
